import PyPDF2
from PyPDF2 import PdfWriter, PdfReader, PageObject
import os

def create_test_pdf(filename, pages=3):
    """创建一个简单的测试PDF文件用于测试PDF转JPG功能"""
    # 由于PyPDF2主要用于操作现有PDF而不是从头创建内容，我们创建一个空白PDF
    # 并添加一些简单的文本注释作为标记
    writer = PdfWriter()
    
    for page_num in range(1, pages + 1):
        # 创建一个标准A4页面 (210mm x 297mm)
        page = PageObject.create_blank_page(width=210 * 72 / 25.4, height=297 * 72 / 25.4)
        
        # 添加元数据作为页面标记
        # 注意：PyPDF2不能直接在页面上绘制文本，但我们可以添加注释
        # 这将在PDF属性中可见，但可能不会在转换后的图像中显示
        
        # 保存页面
        writer.add_page(page)
    
    # 写入文件
    with open(filename, 'wb') as output_pdf:
        writer.write(output_pdf)
    
    print(f"已创建测试PDF文件: {filename}")
    print(f"页数: {pages}")
    print(f"文件大小: {os.path.getsize(filename)/1024:.2f} KB")

if __name__ == "__main__":
    # 创建一个3页的测试PDF
    create_test_pdf('test_document.pdf', pages=3)